package com.leyou.trade.service.impl;

import com.leyou.auth.thread.UserContext;
import com.leyou.common.exceptions.LyException;
import com.leyou.trade.entity.CartItem;
import com.leyou.trade.repository.CartItemRepository;
import com.leyou.trade.service.CartService;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.mongodb.core.query.Update;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * @author 虎哥
 */
@Service
public class CartServiceImpl implements CartService {

    private final CartItemRepository repository;

    private final MongoTemplate template;

    public CartServiceImpl(CartItemRepository repository, MongoTemplate template) {
        this.repository = repository;
        this.template = template;
    }

    @Override
    public void addCartItem(CartItem cartItem) {
        cartItem = isSaveOrUpdate(cartItem);
        // 新增
        repository.save(cartItem);
    }

    private CartItem isSaveOrUpdate(CartItem cartItem) {
        // 获取用户id
        Long userId = UserContext.getUser().getId();
        // 计算购物车条目的id
        String id = buildId(userId, cartItem.getSkuId());
        // 获取num
        Integer num = cartItem.getNum();
        // 尝试取
        Optional<CartItem> optional = repository.findById(id);
        // 判断当前购物车商品是否存在
        if (optional.isPresent()) {
            // 存在
            cartItem = optional.get();
            cartItem.setNum(cartItem.getNum() + num);
        }
        // 补充字段
        cartItem.setId(id);
        cartItem.setUserId(userId);
        cartItem.setUpdateTime(new Date());
        return cartItem;
    }

    @Override
    public List<CartItem> queryCartList() {
        // 获取用户
        Long userId = UserContext.getUser().getId();
        // 根据用户查询
        List<CartItem> list = repository.findByUserId(userId, Sort.by("updateTime").descending());
        if (list == null || list.size() == 0) {
            throw new LyException(404, "购物车为空！");
        }
        return list;
    }

    @Override
    public void updateNum(Long skuId, Integer num) {
        // 获取用户ID
        Long userId = UserContext.getUser().getId();
        // 获取id
        String id = buildId(userId, skuId);
        // 修改
        template.update(CartItem.class)
                .matching(Query.query(Criteria.where("_id").is(id)))
                .apply(Update.update("num", num).set("updateTime", new Date()))
                .first();
    }

    @Override
    public void addCartItemList(List<CartItem> cartItems) {
        List<CartItem> saveOrUpdateList = cartItems
                .stream()
                .map(this::isSaveOrUpdate)
                .collect(Collectors.toList());

        repository.saveAll(saveOrUpdateList);
    }

    private String buildId(Long userId, Long skuId) {
        String pattern = "u%ds%d";
        return String.format(pattern, userId, skuId);
    }
}
